package com.linyaonan.leetcode.medium._430;

/**
 * 你会得到一个双链表，其中包含的节点有一个下一个指针、一个前一个指针和一个额外的 子指针 。这个子指针可能指向一个单独的双向链表，也包含这些特殊的节点。这些子列表可以有一个或多个自己的子列表，以此类推，以生成如下面的示例所示的 多层数据结构 。
 *
 * 给定链表的头节点 head ，将链表 扁平化 ，以便所有节点都出现在单层双链表中。让 curr 是一个带有子列表的节点。子列表中的节点应该出现在扁平化列表中的 curr 之后 和 curr.next 之前 。
 *
 * 返回 扁平列表的 head 。列表中的节点必须将其 所有 子指针设置为 null 。
 *
 * 输入：head = [1,2,3,4,5,6,null,null,null,7,8,9,10,null,null,11,12]
 * 输出：[1,2,3,7,8,11,12,9,10,4,5,6]
 * 解释：输入的多级列表如上图所示。
 *
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/flatten-a-multilevel-doubly-linked-list
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
class Solution {
    /**
     * 整体思路：
     * 1. 当判断到当前节点的child不为空时，记录一下自己的next节点
     * 2. 将child插入到自己的next节点下面，child遍历完成后再与之前的next连接
     * 3. 其中涉及到递归处理所有的child以及内层child
     * @param head
     * @return
     */
    public Node flatten(Node head) {
        if (head == null) {
            return head;
        }
        getEnd(head);
        return head;

    }

    public Node getEnd(Node p) {
        Node pre = null;
        while (p != null) {
            if (p.child != null) {
                Node ppNext = p.next;
                Node end = getEnd(p.child);
                p.next = p.child;
                p.child.prev = p;
                end.next = ppNext;
                if (ppNext != null) {
                    ppNext.prev = end;
                }
                p.child = null;
            }
            pre = p;
            p = p.next;
        }
        return pre;
    }
}